package com.imooc.api.mq;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @version 1.0
 * @Author Mr.Yang
 * @Date 2024-08-22 09:29
 * @question: 生产者方的 消息确认和return回调 不能写到某个方法当中 因为项目初始化的时候 已经初始化了一个rabbitMQ的如消息确认
 *              再写到方法中会报错
 */

@Configuration
@Slf4j
public class RabbitConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * @PostConstruct是Java EE 5 引入的一个注解(Annotation),用于标记在方法上,
     *      以指示该方法是在依赖注入完成后被自动调用的.
     *      它通常用于执行初始化操作,比如资源的初始化、数据的预加载等
     */
    @PostConstruct
    public void initRabbitTemplate() {

        /**
         * 生产者到交换机的消息确认
         */
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {

            /**
             * 回调函数
             * @param correlationData rabbitMQ的确认机制--关联确认(Correlated Confirms)在这种模式下,每个发送的消息都会获得一个唯一的序列号
             * @param ack 交换机是否成功接收到消息 true:成功
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                log.info("进入 confirm.....");
                if (ack) {
                    log.info("交换机成功接收到消息,correlationData: {}", correlationData.getId());
                } else {
                    log.error("交换机接收消息失败,correlationData: {}, cause: {}", correlationData.getId(), cause);
                    //在这里可以添加重试或其他错误处理逻辑
                }
            }
        });

        /**
         * 生产者交换机到队列处的return回调函数
         * 如果成功发送消息 则该方法不会被触发
         */
        rabbitTemplate.setReturnsCallback(new RabbitTemplate.ReturnsCallback() {
            @Override
            public void returnedMessage(ReturnedMessage returnedMessage) {
                log.error("消息被交换机退回,returnedMessage: {}", returnedMessage.toString());
                //在这里可以处理消息被退回的情况
            }
        });
    }


}
